python function for beginners